/*
 * Copyright (c) 2021 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */


#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <string.h>
#include <net/if.h>
#include <arpa/inet.h>
#include <sys/ioctl.h>
#include <netinet/in.h>

#include <securec.h>

#include "local_net_utils.h"

// 加密计算符号等化
#define SHA256_ROTL(a, b) ((((a) >> (32 - (b))) & (0x7fffffff >> (31 - (b)))) | ((a) << (b)))
#define SHA256_SR(a, b) (((a) >> (b)) & (0x7fffffff >> ((b) - 1)))

#define SHA256_Ch(x, y, z) (((x) & (y)) ^ ((~(x)) & (z)))
#define SHA256_Maj(x, y, z) (((x) & (y)) ^ ((x) & (z))^((y) & (z)))
#define SHA256_E0(x) (SHA256_ROTL((x), 30) ^ SHA256_ROTL((x), 19)^SHA256_ROTL((x), 10))
#define SHA256_E1(x) (SHA256_ROTL((x), 26) ^ SHA256_ROTL((x), 21)^SHA256_ROTL((x), 7))
#define SHA256_O0(x) (SHA256_ROTL((x), 25) ^ SHA256_ROTL((x), 14)^SHA256_SR((x), 3))
#define SHA256_O1(x) (SHA256_ROTL((x), 15) ^ SHA256_ROTL((x), 13)^SHA256_SR((x), 10))

/**
 * @brief 计算字符串SHA-256
 *
 * @param str 字符串指针
 * @param length 字符串长度
 * @param sha256 用于保存SHA-256的字符串指针
 * @return int 成功返回0
 */
int  StrSha256(char* str, long length, char* sha256, int size)
{
    const int digit_2 = 2;
    const int digit_4 = 4;
    const int digit_8 = 8;
    const int digit_16 = 16;
    const int digit_64 = 64;
    const int digit_128 = 128;
    const int digit_29 = 29;
    const int digit_56 = 56;

    char *pp, *ppend;
    long l, i, W[64], T1, T2, A, B, C, D, E, F, G, H, H0, H1, H2, H3, H4, H5, H6, H7;
    H0 = 0x6a09e667, H1 = 0xbb67ae85, H2 = 0x3c6ef372, H3 = 0xa54ff53a;
    H4 = 0x510e527f, H5 = 0x9b05688c, H6 = 0x1f83d9ab, H7 = 0x5be0cd19;
    long K[64] = {
        0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5, 0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5,
        0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3, 0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174,
        0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc, 0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da,
        0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7, 0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967,
        0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13, 0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85,
        0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3, 0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070,
        0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5, 0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3,
        0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208, 0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2,
    };
    
    l = length + ((length % digit_64 >= digit_56) ? (digit_128 - length % digit_64) : (digit_64 - length % digit_64));
    if (!(pp = (char*)malloc((unsigned long)l))) {
        return 0;
    }

    for (i = 0; i < length; pp[i + (digit_4 - 1) - digit_2 * (i % digit_4)] = str[i], i++) {
    }

    for (pp[i + (digit_4 - 1) - digit_2 * (i % digit_4)] = digit_128, i++; i < l;
        pp[i + (digit_4 - 1) - digit_2 * (i % digit_4)] = 0, i++) {
    }

    *((long*)(pp + l - digit_4)) = length << (digit_4 - 1);
    *((long*)(pp + l - digit_8)) = length >> digit_29;

    for (ppend = pp + l; pp < ppend; pp += digit_64) {
        for (i = 0; i < digit_16; W[i] = ((long*)pp)[i], i++) {
        }

        for (i = digit_16; i < digit_64; W[i] = (SHA256_O1(W[i - digit_2]) + W[i - (digit_8 - 1)] +
                                           SHA256_O0(W[i - (digit_16 - 1)]) + W[i - digit_16]), i++) {
        }

        A = H0, B = H1, C = H2, D = H3, E = H4, F = H5, G = H6, H = H7;
        for (i = 0; i < digit_64; i++) {
            T1 = H + SHA256_E1(E) + SHA256_Ch(E, F, G) + K[i] + W[i];
            T2 = SHA256_E0(A) + SHA256_Maj(A, B, C);
            H = G, G = F, F = E, E = D + T1, D = C, C = B, B = A, A = T1 + T2;
        }
        H0 += A, H1 += B, H2 += C, H3 += D, H4 += E, H5 += F, H6 += G, H7 += H;
    }
    free(pp - l);
    (void)sprintf_s(sha256, size, "%08lx%08lx%08lx%08lx%08lx%08lx%08lx%08lx", H0, H1, H2, H3, H4, H5, H6, H7);
    return 0;
}

int logFd;
int initLogs(const char *logRoute)
{
	return 0;
  logFd = open(logRoute, O_CREAT | O_APPEND | O_RDWR);
  if (logFd < 0) {
      return -1;
  }
  return logFd;
}

int addLogs_(const char *logs)
{
    if (logFd >= 0 && logs) {
        write(logFd, logs, strlen(logs));
        write(logFd, "\n", strlen("\n"));
        return 0;
    }
    return -1;
}

int closeLogs()
{
    if (logFd >= 0) {
        close(logFd);
        logFd = -1;
    }
    return 0;
}
